// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef NET_SOCKET_TCP_SERVER_SOCKET_H_
#define NET_SOCKET_TCP_SERVER_SOCKET_H_

#include <memory>

#include "base/macros.h"
#include "net/base/ip_endpoint.h"
#include "net/base/net_export.h"
#include "net/log/net_log.h"
#include "net/socket/server_socket.h"
#include "net/socket/tcp_socket.h"

namespace net {

class NET_EXPORT TCPServerSocket : public ServerSocket {
public:
    TCPServerSocket(NetLog* net_log, const NetLog::Source& source);
    ~TCPServerSocket() override;

    // net::ServerSocket implementation.
    int Listen(const IPEndPoint& address, int backlog) override;
    int GetLocalAddress(IPEndPoint* address) const override;
    int Accept(std::unique_ptr<StreamSocket>* socket,
        const CompletionCallback& callback) override;

    // Detachs from the current thread, to allow the socket to be transferred to
    // a new thread. Should only be called when the object is no longer used by
    // the old thread.
    void DetachFromThread();

private:
    // Converts |accepted_socket_| and stores the result in
    // |output_accepted_socket|.
    // |output_accepted_socket| is untouched on failure. But |accepted_socket_| is
    // set to NULL in any case.
    int ConvertAcceptedSocket(
        int result,
        std::unique_ptr<StreamSocket>* output_accepted_socket);
    // Completion callback for calling TCPSocket::Accept().
    void OnAcceptCompleted(std::unique_ptr<StreamSocket>* output_accepted_socket,
        const CompletionCallback& forward_callback,
        int result);

    TCPSocket socket_;

    std::unique_ptr<TCPSocket> accepted_socket_;
    IPEndPoint accepted_address_;
    bool pending_accept_;

    DISALLOW_COPY_AND_ASSIGN(TCPServerSocket);
};

} // namespace net

#endif // NET_SOCKET_TCP_SERVER_SOCKET_H_
